package org.yun.common.utils.webeditor;

import org.springframework.web.util.HtmlUtils;

import java.beans.PropertyEditorSupport;

/**
 * HTML清理
 */
public class HtmlEscapeEditor extends PropertyEditorSupport {

    /**
     * 是否移除两端空白
     */
    private boolean trim;

    /**
     * 是否将空转换为null
     */
    private boolean emptyAsNull;

    /**
     * @param trim        是否移除两端空白
     * @param emptyAsNull 是否将空转换为null
     */
    public HtmlEscapeEditor(boolean trim, boolean emptyAsNull) {
        this.trim = trim;
        this.emptyAsNull = emptyAsNull;
    }

    /**
     * 获取内容
     *
     * @return 内容
     */
    @Override
    public String getAsText() {
        Object value = getValue();
        return value != null ? value.toString() : "";
    }

    /**
     * 设置内容
     *
     * @param text 内容
     */
    @Override
    public void setAsText(String text) {
        if (text != null) {
            String value = trim ? text.trim() : text;
            value = HtmlUtils.htmlEscape(value);
            if (emptyAsNull && "".equals(value)) {
                value = null;
            }
            setValue(value);
        } else {
            setValue(null);
        }
    }
}
